from pydantic import BaseModel
from typing import Dict

class Choice(BaseModel):
    index: int
    message: Dict[str, str]
    finish_reason: str
